package cn.edu.xjtu.work.deleteAndEarn;

import java.util.Arrays;

/**
 * 740. 删除并获得点数
 * 
 * 给你一个整数数组 nums ，你可以对它进行一些操作。
 * 
 * 每次操作中，选择任意一个 nums[i] ，删除它并获得 nums[i] 的点数。之后，你必须删除 所有 等于 nums[i] - 1 和 nums[i]
 * + 1 的元素。
 */
public class Solution {
    public static void main(String[] args) {
        Solution solu = new Solution();
        int[] nums = { 3, 1 };
        System.out.println(solu.deleteAndEarn(nums));
    }

    public int deleteAndEarn(int[] nums) {
        int maxVal = 0;
        for (int i : nums) {
            maxVal = Math.max(maxVal, i);
        }
        int[] arr = new int[maxVal + 1];
        for (int i : nums) {
            arr[i] = arr[i] + i;
        }
        System.out.println(Arrays.toString(arr));
        return rob(arr);
    }

    public int rob(int[] nums) {
        if (nums.length == 1) {
            return nums[0];
        }
        if (nums.length == 2) {
            return Math.max(nums[0], nums[1]);
        }
        int first = nums[0];
        int second = Math.max(nums[0], nums[1]);
        for (int i = 2; i < nums.length; i++) {
            int temp = Math.max(first + nums[i], second);
            first = second;
            second = temp;
        }
        return second;
    }
}
